package com.atguigu.spzx.manager.utils;

import com.atguigu.spzx.model.entity.system.SysMenu;

import java.util.ArrayList;
import java.util.List;

/**
 * 作者:hfj
 * 功能:
 * 日期: 2024/3/28 10:52
 */
public class MenuHelper {

    //递归实现
    public static List<SysMenu> buildTree(List<SysMenu> sysMenuList){
        //通过递归完成封装过程
        //创建一个集合用于封装并返回最终数据
        List<SysMenu> trees = new ArrayList<>();

        //遍历所有菜单集合
        for (SysMenu sysMenu:sysMenuList){
            //找到递归操作入口,第一层菜单
            if (sysMenu.getParentId().longValue() == 0){
                trees.add(findChildren(sysMenu,sysMenuList));
            }
        }
        return trees;
    }

    //递归查找子级菜单
    private static SysMenu findChildren(SysMenu sysMenu, List<SysMenu> sysMenuList) {
        sysMenu.setChildren(new ArrayList<>());
        //递归查询
        for (SysMenu it : sysMenuList){
            //判断Id和ParentId是否相同
            if (sysMenu.getId().longValue() == it.getParentId().longValue()){
                //it就是下一层数据,进行封装
                sysMenu.getChildren().add(findChildren(it,sysMenuList));
            }
        }
        return sysMenu;
    }
}
